home *** CD-ROM | disk | FTP | other *** search
/ Aminet 2 / Aminet AMIGA CDROM (1994)(Walnut Creek)[Feb 1994][W.O. 44790-1].iso / Aminet / util / gnu / textutils_1_3.LHA / textutils-1.3 / src / fold.c < prev    next >
C/C++ Source or Header  |  1992-06-29  |  6KB  |  251 lines
  1. /* fold -- wrap each input line to fit in specified width.
  2.    Copyright (C) 1991 Free Software Foundation, Inc.
  3.  
  4.    This program is free software; you can redistribute it and/or modify
  5.    it under the terms of the GNU General Public License as published by
  6.    the Free Software Foundation; either version 2, or (at your option)
  7.    any later version.
  8.  
  9.    This program is distributed in the hope that it will be useful,
  10.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.    GNU General Public License for more details.
  13.  
  14.    You should have received a copy of the GNU General Public License
  15.    along with this program; if not, write to the Free Software
  16.    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.  */
  17.  
  18. /* Written by David MacKenzie. */
  19.  
  20. #define _GNU_SOURCE
  21. #include <ctype.h>
  22. #ifndef isblank
  23. #define isblank(c) ((c) == ' ' || (c) == '\t')
  24. #endif
  25. #include <stdio.h>
  26. #include <getopt.h>
  27. #include <sys/types.h>
  28. #include "system.h"
  29.  
  30. char *xrealloc ();
  31. int adjust_column ();
  32. int fold_file ();
  33. void error ();
  34.  
  35. /* If nonzero, try to break on whitespace. */
  36. int break_spaces;
  37.  
  38. /* If nonzero, count bytes, not column positions. */
  39. int count_bytes;
  40.  
  41. /* If nonzero, at least one of the files we read was standard input. */
  42. int have_read_stdin;
  43.  
  44. /* The name this program was run with. */
  45. char *program_name;
  46.  
  47. struct option longopts[] =
  48. {
  49.   {"bytes", 0, NULL, 'b'},
  50.   {"spaces", 0, NULL, 's'},
  51.   {"width", 1, NULL, 'w'},
  52.   {NULL, 0, NULL, 0}
  53. };
  54.  
  55. void
  56. main (argc, argv)
  57.      int argc;
  58.      char **argv;
  59. {
  60.   int width = 80;
  61.   int i;
  62.   int optc;
  63.   int errs = 0;
  64.  
  65.   program_name = argv[0];
  66.   break_spaces = count_bytes = have_read_stdin = 0;
  67.  
  68.   while ((optc = getopt_long (argc, argv, "bsw:", longopts, (int *) 0))
  69.      != EOF)
  70.     {
  71.       switch (optc)
  72.     {
  73.     case 'b':        /* Count bytes rather than columns. */
  74.       count_bytes = 1;
  75.       break;
  76.  
  77.     case 's':        /* Break at word boundaries. */
  78.       break_spaces = 1;
  79.       break;
  80.  
  81.     case 'w':        /* Line width. */
  82.       width = atoi (optarg);
  83.       if (width < 1)
  84.         error (1, 0, "%s: invalid line width", optarg);
  85.       break;
  86.  
  87.     default:
  88.       fprintf (stderr, "\
  89. Usage: %s [-bs] [-w width] [--bytes] [--spaces] [--width=width] [file...]\n",
  90.            argv[0]);
  91.       exit (1);
  92.     }
  93.     }
  94.  
  95.   if (argc == optind)
  96.     errs |= fold_file ("-", width);
  97.   else
  98.     for (i = optind; i < argc; i++)
  99.       errs |= fold_file (argv[i], width);
  100.  
  101.   if (have_read_stdin && fclose (stdin) == EOF)
  102.     error (1, errno, "-");
  103.   if (fclose (stdout) == EOF)
  104.     error (1, errno, "write error");
  105.  
  106.   exit (errs);
  107. }
  108.  
  109. /* Fold file FILENAME, or standard input if FILENAME is "-",
  110.    to stdout, with maximum line length WIDTH.
  111.    Return 0 if successful, 1 if an error occurs. */
  112.  
  113. int
  114. fold_file (filename, width)
  115.      char *filename;
  116.      int width;
  117. {
  118.   FILE *istream;
  119.   register int c;
  120.   int column = 0;        /* Screen column where next char will go. */
  121.   int offset_out = 0;        /* Index in `line_out' for next char. */
  122.   static char *line_out = NULL;
  123.   static size_t allocated_out = 0;
  124.  
  125.   if (!strcmp (filename, "-"))
  126.     {
  127.       istream = stdin;
  128.       have_read_stdin = 1;
  129.     }
  130.   else
  131.     istream = fopen (filename, "r");
  132.  
  133.   if (istream == NULL)
  134.     {
  135.       error (0, errno, "%s", filename);
  136.       return 1;
  137.     }
  138.  
  139.   while ((c = getc (istream)) != EOF)
  140.     {
  141.       if (offset_out + 1 >= allocated_out)
  142.     {
  143.       allocated_out += 1024;
  144.       line_out = xrealloc (line_out, allocated_out);
  145.     }
  146.       
  147.       if (c == '\n')
  148.     {
  149.       line_out[offset_out++] = c;
  150.       fwrite (line_out, sizeof (char), offset_out, stdout);
  151.       column = offset_out = 0;
  152.       continue;
  153.     }
  154.  
  155.     rescan:
  156.       column = adjust_column (column, c);
  157.  
  158.       if (column >= width)
  159.     {
  160.       /* This character would make the line too long.
  161.          Print the line plus a newline, and make this character
  162.          start the next line. */
  163.       if (break_spaces)
  164.         {
  165.           /* Look for the last blank. */
  166.           int logical_end;
  167.  
  168.           for (logical_end = offset_out - 1; logical_end >= 0;
  169.            logical_end--)
  170.         if (isblank (line_out[logical_end]))
  171.           break;
  172.           if (logical_end >= 0)
  173.         {
  174.           int i;
  175.  
  176.           /* Found a blank.  Don't output the part after it. */
  177.           logical_end++;
  178.           fwrite (line_out, sizeof (char), logical_end, stdout);
  179.           putchar ('\n');
  180.           /* Move the remainder to the beginning of the next line.
  181.              The areas being copied here might overlap. */
  182.           bcopy (line_out + logical_end, line_out,
  183.              offset_out - logical_end);
  184.           offset_out -= logical_end;
  185.           for (column = i = 0; i < offset_out; i++)
  186.             column = adjust_column (column, line_out[i]);
  187.           goto rescan;
  188.         }
  189.         }
  190.       line_out[offset_out++] = '\n';
  191.       fwrite (line_out, sizeof (char), offset_out, stdout);
  192.       column = offset_out = 0;
  193.       goto rescan;
  194.     }
  195.  
  196.       line_out[offset_out++] = c;
  197.     }
  198.  
  199.   if (offset_out)
  200.     fwrite (line_out, sizeof (char), offset_out, stdout);
  201.  
  202.   if (ferror (istream))
  203.     {
  204.       error (0, errno, "%s", filename);
  205.       if (strcmp (filename, "-"))
  206.     fclose (istream);
  207.       return 1;
  208.     }
  209.   if (strcmp (filename, "-") && fclose (istream) == EOF)
  210.     {
  211.       error (0, errno, "%s", filename);
  212.       return 1;
  213.     }
  214.  
  215.   if (ferror (stdout))
  216.     {
  217.       error (0, errno, "write error");
  218.       return 1;
  219.     }
  220.  
  221.   return 0;
  222. }
  223.  
  224. /* Assuming the current column is COLUMN, return the column that
  225.    printing C will move the cursor to.
  226.    The first column is 0. */
  227.  
  228. int
  229. adjust_column (column, c)
  230.      int column;
  231.      char c;
  232. {
  233.   if (!count_bytes)
  234.     {
  235.       if (c == '\b')
  236.     {
  237.       if (column > 0)
  238.         column--;
  239.     }
  240.       else if (c == '\r')
  241.     column = 0;
  242.       else if (c == '\t')
  243.     column = column + 8 - column % 8;
  244.       else /* if (isprint (c)) */
  245.     column++;
  246.     }
  247.   else
  248.     column++;
  249.   return column;
  250. }
  251.